#include "../common/elasticsearch.hpp"
#include <gflags/gflags.h>

DEFINE_bool(run_mode, false, "程序的运行模式, false-调试模式, true-发布模式");
DEFINE_string(log_file, "", "发布模式下, 用于指定日志文件的路径");
DEFINE_int32(log_level, 0, "发布模式下, 用于指定日志的输出等级");

int main(int argc, char* argv[]) 
{
    google::ParseCommandLineFlags(&argc, &argv, false);
    Init_logger(FLAGS_run_mode, FLAGS_log_file, FLAGS_log_level);

    std::vector<std::string> host_list = {"http://127.0.0.1:9200/"};
    auto client = std::make_shared<elasticlient::Client>(host_list);

    bool ret = ESIndex(client, "test_user").append("nickname")
        .append("phone", "keyword", "standard", true)
        .create();
    if (ret == false) {
        LOG_ERROR("索引创建失败!");
        return -1;
    } else {
        LOG_INFO("索引创建成功!");
    }

    // 数据的新增
    ret = ESInsert(client, "test_user")
        .append("nickname", "李四")
        .append("phone", "13912345678")
        .insert("00001");
    if (ret == false) {
        LOG_ERROR("数据插入失败!");
        return -1;
    } else {
        LOG_INFO("数据插入成功!");
    }

    // 数据的修改
    ret = ESInsert(client, "test_user")
        .append("nickname", "李四")
        .append("phone", "12482348923")
        .insert("00001");
    if (ret == false) {
        LOG_ERROR("数据更新失败!");
        return -1;
    } else {
        LOG_INFO("数据更新成功!");
    }

    // 数据的查询
    Json::Value user = ESSearch(client, "test_user")
        .append_should_match("phone.keyword", "12482348923")
        // .append_must_not_terms("nickname.keyword", {"李四"})
        .search();
    if(user.empty() || user.isArray() == false) {
        LOG_ERROR("结果为空，或者结果不是数组!");
        return -1;
    } 
    int sz = user.size();
    for(int i = 0; i < sz; ++i) {
        LOG_INFO("nickname: {}", user[i]["_source"]["nickname"].asString());
    }

    // 数据的删除
    ret = ESRemove(client, "test_user").remove("00001");
    if(ret == false) {
        LOG_ERROR("数据删除失败!");
        return -1;
    } else {
        LOG_INFO("数据删除成功!");
    }

    return 0;
}